Optimise Kotlin code for button clicked state